package com.ck.util;

import org.springframework.util.ResourceUtils;

import java.io.*;
import java.util.zip.CRC32;
import java.util.zip.CheckedOutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * zip工具类
 *
 * @author linqiang
 * @date 2018-08-11 12:03
 */
public class ZipCompressUtils {

    private static final int BUFFER = 8192;

    private File zipFile;

    public ZipCompressUtils(String pathName) {
        zipFile = new File(pathName);
    }

    public static void main(String[] args) throws FileNotFoundException {
        //获取跟目录
        File path = new File(ResourceUtils.getURL("classpath:").getPath());
        if (!path.exists()) {
            path = new File("");
        }
        System.out.println("path:" + path.getAbsolutePath());
        //如果上传目录为/static/images/upload/，则可以如下获取：
//        ZipCompressUtils zc = new ZipCompressUtils("E:/resource.zip");
//        zc.compress(path.getAbsolutePath() + "/static/threescreen");
    }

    /**
     * 压缩文件或文件夹
     *
     * @param pathName 文件或文件夹路径
     */
    public void compress(String... pathName) {
        ZipOutputStream out = null;
        try {
            FileOutputStream fileOutputStream = new FileOutputStream(zipFile);
            CheckedOutputStream cos = new CheckedOutputStream(fileOutputStream,
                    new CRC32());
            out = new ZipOutputStream(cos);
            String basedir = "";
            for (String aPathName : pathName) {
                compress(new File(aPathName), out, basedir);
            }
            out.close();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 压缩文件或文件夹
     *
     * @param srcPathName 文件或文件夹路径
     */
    public void compress(String srcPathName) {
        File file = new File(srcPathName);
        if (!file.exists()) {
            throw new RuntimeException(srcPathName + "不存在！");
        }
        try {
            FileOutputStream fileOutputStream = new FileOutputStream(zipFile);
            CheckedOutputStream cos = new CheckedOutputStream(fileOutputStream,
                    new CRC32());
            ZipOutputStream out = new ZipOutputStream(cos);
            String basedir = "";
            compress(file, out, basedir);
            out.close();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 压缩文件或文件夹
     *
     * @param file    文件或文件夹
     * @param out     zip输出流
     * @param basedir 压缩后的基础路径
     */
    private void compress(File file, ZipOutputStream out, String basedir) {
        /* 判断是目录还是文件 */
        if (file.isDirectory()) {
            System.out.println("压缩：" + basedir + file.getName());
            this.compressDirectory(file, out, basedir);
        } else {
            System.out.println("压缩：" + basedir + file.getName());
            this.compressFile(file, out, basedir);
        }
    }

    /**
     * 压缩一个目录
     *
     * @param dir     压缩后的文件路径
     * @param out     输出了
     * @param basedir 输出到的基础路径
     */
    private void compressDirectory(File dir, ZipOutputStream out, String basedir) {
        if (!dir.exists()) {
            return;
        }

        File[] files = dir.listFiles();
        if (null != files) {
            for (File file : files) {
                //递归
                compress(file, out, basedir + dir.getName() + "/");
            }
        }
    }

    /**
     * 压缩一个文件
     *
     * @param file    压缩后的文件路径
     * @param out     输出了
     * @param basedir 输出到的基础路径
     */
    private void compressFile(File file, ZipOutputStream out, String basedir) {
        if (!file.exists()) {
            return;
        }
        try {
            BufferedInputStream bis = new BufferedInputStream(
                    new FileInputStream(file));
            ZipEntry entry = new ZipEntry(basedir + file.getName());
            out.putNextEntry(entry);
            int count;
            byte[] data = new byte[BUFFER];
            while ((count = bis.read(data, 0, BUFFER)) != -1) {
                out.write(data, 0, count);
            }
            bis.close();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }
}
